#!/usr/bin/env bash

# train
python3.6 main.py \
ctdet \
--exp_id coco_dla_2x \
--batch_size 128 \
--master_batch 9 \
--lr 5e-4 \
--gpus 0,1,2,3,4,5,6,7 \
--num_workers 16 \
--num_epochs 230 \
lr_step 180,210

# or use the following command if your have coco_s2_dla_1x trained
# python3.6 main.py \
# ctdet \
# --exp_id coco_dla_2x \
# --batch_size 128 \
# --master_batch 9 \
# --lr 5e-4 \
# --gpus 0,1,2,3,4,5,6,7 \
# --num_workers 16 \
# --load_model ../exp/ctdet/coco_dla_1x/model_90.pth \
# --resume

# test
python3.6 test.py \
ctdet \
--exp_id coco_dla_2x \
--keep_res \
--resume

# flip test
python3.6 test.py \
ctdet \
--exp_id coco_dla_2x \
--keep_res \
--resume \
--flip_test

# multi scale test
python3.6 test.py \
ctdet \
--exp_id coco_dla_2x \
--keep_res \
--resume \
--flip_test \
--test_scales 0.5,0.75,1,1.25,1.5
